import type { PageResult } from '@vben/request';

import { requestClient, requestStream } from '#/api/request';

export namespace OllamaModelApi {
  export interface OllamaModel {
    name: string; // 名称
    size: number; // 大小
    apiUrl?: string; // 地址
    modified_at?: string; // 创建时间
    hasJoinModel?: number; // 是否加入模型
  }
}

// 获取已安装大模型列表
export function getLargeModels(params: any) {
  return requestClient.get<PageResult<OllamaModelApi.OllamaModel>>(
    '/ai/v1/ollama/list',
    { params },
  );
}

// 安装大模型
export function installModel() {
  return requestStream('/ai/v1/ollama/installModel', {});
}
// 删除大模型
export function deleteModel(params: any) {
  return requestClient.get<any>('/ai/v1/ollama/deleteModel', { params });
}
